Summary

• Adds IsSticky() bool to the SlotReservationInfo interface so that custom SlotSupplier implementations can distinguish sticky vs normal workflow task queue polls at slot reservation time.
• For mixed-mode pollers (the default), the sticky/normal decision is moved before ReserveSlot so IsSticky() is accurate for all poller configurations.
• For activity, local activity, and nexus slots, IsSticky() always returns false.

Fixes #2191

Approach

Commit 1 -- Add IsSticky() to the public interface:

• Extends SlotReservationInfo with IsSticky() bool
• Propagates isSticky through slotReservationData and trackingSlotSupplier
• Sets the flag based on poller mode in runPoller

Commit 2 -- Make IsSticky() accurate for mixed-mode pollers:

• Introduces a pollHint struct to carry a pre-decided sticky/normal choice
• Adds decideNextPollKind() on workflowTaskPoller which commits to sticky vs normal before slot reservation (using the same backlog/pending-count balancing logic)
• Modifies getNextPollRequest() to honor the pre-decided hint, skipping the inline re-decision
• Threads the hint through taskPoller.PollTask() -> poll() -> getNextPollRequest()
• Handles cancellation: undoPollDecision() reverses the counter increment if reservation fails or is cancelled
• Expands test coverage with 12 new tests for counter lifecycle, hint propagation, and all poller modes

Test plan

• All 298 existing internal package tests pass
• New unit tests for decideNextPollKind() across all modes (Sticky, NonSticky, Mixed with balancing, backlog, disabled cache)
• New unit tests for undoPollDecision() (Mixed and non-Mixed)
• New unit tests for getNextPollRequest() with and without hint
• Counter lifecycle tests (happy path: decide -> poll -> release; cancel path: decide -> fail -> undo)
• go vet and linter clean

Made with Cursor